package cn.edu.xjtu.carlWay.dynamicProgramming.partitionEqualSubsetSum;

import java.util.Arrays;

/**
 * 416. 分割等和子集
 * 给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。
 * <p>
 * https://leetcode-cn.com/problems/partition-equal-subset-sum/
 */
public class Solution_1d_DP {
    public static void main(String[] args) {
        Solution_1d_DP solution = new Solution_1d_DP();
        int[] nums = {1, 5, 11, 5};
        System.out.println(solution.canPartition(nums));
    }

    public boolean canPartition(int[] nums) {
        int sum = Arrays.stream(nums).sum();
        if (sum % 2 != 0) {
            return false;
        }
        int[] dp = new int[sum / 2 + 1];
        for (int i = 0; i < nums.length; i++) {
            for (int j = dp.length - 1; j >= 0; j--) {
                if (j - nums[i] >= 0) {
                    dp[j] = Math.max(dp[j], dp[j - nums[i]] + nums[i]);
                }
            }
        }

        return dp[sum / 2] == sum / 2;
    }
}
